// created by noellynch
// Jun 3, 2009

package com.sceneworker.jmetreeextensions;

import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;

import javax.swing.Icon;
import javax.swing.ImageIcon;

import com.acarter.composabletree.ComposableTreeNode;
import com.jme.image.Texture;
import com.jme.scene.state.TextureState;

public class SceneWorkerTextureTreeNode implements ComposableTreeNode {

	private HashMap<String, ImageIcon> iconCache = new HashMap<String, ImageIcon>();

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChild(java.lang.Object,
	 *      int)
	 */
	public Object getChild(Object parent, int index) {
		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			Texture texture = null;
			int found = -1;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				texture = state.getTexture(unit);

				if(texture != null)
					found++;

				if(found == index)
					break;
			}

			return texture;
		}
		else
			return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildCount(java.lang.Object)
	 */
	public int getChildCount(Object parent) {
		int count = 0;

		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				if(state.getTexture(unit) != null)
					count++;
			}
		}

		return count;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildIndex(java.lang.Object,
	 *      java.lang.Object)
	 */
	public int getChildIndex(Object parent, Object child) {
		int index = -1;

		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				Texture texture = state.getTexture(unit);

				if(texture != null)
					index++;

				if(texture.equals(child))
					break;
			}
		}

		return index;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getParent(java.lang.Object)
	 */
	public Object getParent(Object child) {

		// Textures are not aware of their parent
		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeClassType()
	 */
	public Class<?> getNodeClassType() {

		return Texture.class;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getIcon(Object)
	 */
	public Icon getIcon(Object node) {

		if(node instanceof Texture) {

			String location = ((Texture) node).getImageLocation();

			if(location != null) {

				if(iconCache.containsKey(location))
					return iconCache.get(location);

				try {
					Image image = new ImageIcon(new URL(location)).getImage();

					BufferedImage texture = new BufferedImage(12, 12, BufferedImage.TYPE_INT_RGB);
					Graphics2D graphics2D = texture.createGraphics();
					graphics2D.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
					graphics2D.drawImage(image, 0, 0, 12, 12, null);

					ImageIcon icon = new ImageIcon(texture);
					iconCache.put(location, icon);
					
					return icon;
				}
				catch (MalformedURLException e) {
					// consume it
				}
			}
		}

		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeText(Object,
	 *      boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		if(node instanceof Texture) {

			String location = ((Texture) node).getImageLocation();

			if(location != null) {
				File file = new File(location);

				if(file != null)
					return file.getName();
			}

			String className = node.getClass().getSimpleName();

			if(className.length() == 0)
				className = node.getClass().getName();

			return className;
		}

		return (node != null) ? node.toString() : "";
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeToolTipText(Object,
	 *      boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeToolTipText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		return (node != null) ? node.toString() : "";
	}

}

